package com.LeetCode.Learning;

import java.util.HashSet;
import java.util.Set;

/**
 * 在英语中，我们有一个叫做 词根(root) 的概念，可以词根后面添加其他一些词组成另一个较长的单词——我们称这个词为 继承词(successor)。
 * 例如，词根an，跟随着单词 other(其他)，可以形成新的单词 another(另一个)。
 *
 * 现在，给定一个由许多词根组成的词典 dictionary 和一个用空格分隔单词形成的句子 sentence。
 * 你需要将句子中的所有继承词用词根替换掉。如果继承词有许多可以形成它的词根，则用最短的词根替换它。
 *
 * 你需要输出替换之后的句子。
 *
 *
 *
 * 示例 1：
 *
 * 输入：dictionary = ["cat","bat","rat"], sentence = "the cattle was rattled by the battery"
 * 输出："the cat was rat by the bat"
 * 示例 2：
 *
 * 输入：dictionary = ["a","b","c"], sentence = "aadsfasf absbs bbab cadsfafs"
 * 输出："a a b c"
 *
 *
 * 提示：
 *
 * 1 <= dictionary.length <= 1000
 * 1 <= dictionary[i].length <= 100
 * dictionary[i] 仅由小写字母组成。
 * 1 <= sentence.length <= 10^6
 * sentence 仅由小写字母和空格组成。
 * sentence 中单词的总量在范围 [1, 1000] 内。
 * sentence 中每个单词的长度在范围 [1, 1000] 内。
 * sentence 中单词之间由一个空格隔开。
 * sentence 没有前导或尾随空格。
 *该解决方案使用 HashSet 存储词根，然后针对每个单词找到最短的词根并进行替换，最后返回拼接好的句子。你可以根据需要进行适当的修改和调整。
 */

public class ReplaceWords {
    public static String replaceWords(String[] dictionary, String sentence) {
        Set<String> rootSet = new HashSet<>();
        for (String root : dictionary) {
            rootSet.add(root);
        }

        StringBuilder sb = new StringBuilder();
        String[] words = sentence.split("\\s+");//按任意空白字符切分
        for (String word : words) {
            String prefix = "";
            for (int i = 1; i <= word.length(); i++) {
                prefix = word.substring(0, i);
                if (rootSet.contains(prefix)) {
                    break;
                }
            }
            sb.append(" ").append(prefix);
        }

        return sb.substring(1).toString();
    }

    public static void main(String[] args) {
        String[] dictionary = {"cat", "bat", "rat"};
        String sentence = "the cattle was rattled by the battery";
        System.out.println("Replaced sentence: " + replaceWords(dictionary, sentence));

        String[] dictionary2 = {"a", "b", "c"};
        String sentence2 = "aadsfasf absbs bbab cadsfafs";
        System.out.println("Replaced sentence: " + replaceWords(dictionary2, sentence2));
    }
}

